from common.account.models import (
    Information,
    DWCDark2Information,
    DWCDark3Information,
    DWCDark4Information,
    DWCTaiTanInformation,
    Device,
    DWCDark2Device,
    DWCDark3Device,
    DWCDark4Device,
    DWCTaiTanDevice,
)

ACCOUNT_MODEL_ROUTER = {
    'zs': Information,
    'dark2': DWCDark2Information,
    'dark3': DWCDark3Information,
    'dark4': DWCDark4Information,
    'tt': DWCTaiTanInformation
}

DEVICE_MODEL_ROUTER = {
    'zs': Device,
    'dark2': DWCDark2Device,
    'dark3': DWCDark3Device,
    'dark4': DWCDark4Device,
    'tt': DWCTaiTanDevice,
}
